Fixed bug on my part ... :/