still trying to solve bug