Two lines of code and it works now.