Did a quick look over the code and added a quick fix