Bug fix, something went missing