Improve errorhandling