changed it to get it working