Fix wrong return value