Should fix reference error