Fix import errors introduced with new styling module