fixed an issue :/