Fixed my own stupidity (for now)