Trying to fix this. Still not working..