quick fix and it looks like it works!