Temp bug fix