Small update
Fixed wrong return