Fixing onPlayCallback