Fix bug. Get some sleep nick