Fix small issue with return results