fixed bug in code by returning