Fixes view and preview buttons on unit page

@cahrens @andy-armstrong 
I had no idea that we had to do this!