Add return statement for error