Maybe typehint class instead of just comment ?