Add return checker