Fix: forgot to add Window class